from pydantic import BaseModel
from typing import List

from app.schema.response import BaseResponse


class DetectionResult(BaseModel):
    class_id: int
    class_name: str
    confidence: float
    xmin: int
    ymin: int
    xmax: int
    ymax: int


class Base64Request(BaseModel):
    image_base64: str


class DetectionResponse(BaseResponse):
    data: List[DetectionResult] = []
